package dao

import (
	"gorm.io/driver/mysql"
	"gorm.io/gorm"
)

//定义数据库全局变量
var (
	DB *gorm.DB
)

//InitMySQL 初始化数据库
func InitMySQL() (err error) {
	dsn := "root:root@tcp(127.0.0.1:53306)/db1?charset=utf8mb4&parseTime=True&loc=Local"
	DB, err = gorm.Open(mysql.Open(dsn), &gorm.Config{})
	if err != nil {
		return
	}
	db, _ := DB.DB()
	return db.Ping()
}

//Close 断开数据库连接
func Close() {
	db, _ := DB.DB()
	db.Close()
}
